# 作业
# 不限次数输入，选择3退出
# 1.注册
#   如果有这个用户，就不用再注册
#   用字典name_dict = {"root":{"age":18,"passwd":"123456"},\
#                   "admin":{"age":19,"passwd":"admin123"}}
#   输入用户名、年龄、密码
# 2.查询
#   输入要查询的姓名，展示这个人的详细信息
# 3.退出
# 后续考虑：写入文件
# json库

# 导入json库
import json

# 定义一个空字典，存放注册信息
name_dict = {}
# 定义注册函数
def register():
    name = input("please input your name:")
    if name in name_dict:
        print("用户已注册")
    else:
        age = input("please input your age:")
        passwd = input("please input your passwd:")
        name_dict[name] = {"age": age, "passwd": passwd}
        print("注册完成")

# 定义查询函数
def query():
    name = input("please input the name:")
    if name in name_dict:
        print("name is:", name)
        print("age is:", name_dict[name]["age"])
        print("passwd is:", name_dict[name]["passwd"])
    else:
        print("该用户未注册")

# 定义保存字典数据的文件
# 利用json.dump()把dict数据转换成字符串并存储在定义的json文件中
def save_dict_file():
    with open("register.json", "w") as f:
        json.dump(name_dict, f)

# 循环
while True:
    num = input("please input a number(1.注册 2.查询 3.退出):")
    if num == "1":
        register()
    elif num == "2":
        query()
    elif num == "3":
        save_dict_file()
        print("exit")
        break
    else:
        print("输入错误")
